Skip to content

Feature/jax latent#1154

Merged
Jammy2211 merged 6 commits intofeature/jax_wrapperfrom
feature/jax_latent
Oct 6, 2025
Merged

Feature/jax latent#1154
Jammy2211 merged 6 commits intofeature/jax_wrapperfrom
feature/jax_latent

Conversation

@Jammy2211
Copy link
Copy Markdown
Collaborator

Update the compute_latent_variables method to support:

  • Jax jit and vmap inside of it, which required it to not return a dict and thus the keys of the latent variables became a class variables LATENT_KEYS.
  • Make it so that just a latent.summary file can be created via PDF redraws to speed up latent variable time computation.

@Jammy2211 Jammy2211 requested a review from rhayes777 October 3, 2025 09:10
@Jammy2211 Jammy2211 merged commit 202b36a into feature/jax_wrapper Oct 6, 2025
0 of 4 checks passed
@Jammy2211 Jammy2211 deleted the feature/jax_latent branch November 30, 2025 16:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants